/*
 * Copyright (c) Huawei Technologies Co., Ltd. 2023. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#ifndef ACLNN_ADAPTER_PLUGIN_UTILS_H
#define ACLNN_ADAPTER_PLUGIN_UTILS_H

#include "atb/atb_infer.h"
#include "atb/operation.h"
#include "utils.h"

namespace atb_plugin {

const int DIM0 = 0;
const int DIM1 = 1;
const int DIM2 = 2;
const int DIM3 = 3;
const int NUM0 = 0;
const int NUM1 = 1;
const int NUM2 = 2;
const int NUM3 = 3;
const int NUM4 = 4;
const int NUM5 = 5;
const int NUM6 = 6;

atb::SVector<int64_t> GetCopyTensorStride(atb::Dims &tensorDims);

atb::Tensor SqueezeBatchSeq(atb::Tensor atbTensor);

} // namespace atb_plugin
#endif